﻿using System;
using System.Diagnostics;
using System.Globalization;
using System.Runtime.InteropServices;
using System.ComponentModel.Design;
using Microsoft.Win32;
using Microsoft.VisualStudio;
using Microsoft.VisualStudio.Shell.Interop;
using Microsoft.VisualStudio.OLE.Interop;
using Microsoft.VisualStudio.Shell;
using Expect.Vs2008.Extensions.LanguageService;
using Expect.VisualStudio.Extensions;

namespace Expect.Vs2008.Extensions
{
    /// <summary>
    /// This is the class that implements the package exposed by this assembly.
    ///
    /// The minimum requirement for a class to be considered a valid package for Visual Studio
    /// is to implement the IVsPackage interface and register itself with the shell.
    /// This package uses the helper classes defined inside the Managed Package Framework (MPF)
    /// to do it: it derives from the Package class that provides the implementation of the 
    /// IVsPackage interface and uses the registration attributes defined in the framework to 
    /// register itself and its components with the shell.
    /// </summary>
    // This attribute tells the PkgDef creation utility (CreatePkgDef.exe) that this class is
    // a package.
    [PackageRegistration(UseManagedResourcesOnly = true)]
    // A Visual Studio component can be registered under different registry roots; for instance
    // when you debug your package you want to register it in the experimental hive. This
    // attribute specifies the registry root to use if no one is provided to regpkg.exe with
    // the /root switch.
    [DefaultRegistryRoot("Software\\Microsoft\\VisualStudio\\9.0Exp")]

    // This attribute is used to register the informations needed to show the this package
    // in the Help/About dialog of Visual Studio.
    [InstalledProductRegistration(true, "Expect framework", "", Guids.guidExpect_ExtensionsPkgString)]
    //[InstalledProductRegistration("#110", "#112", "1.0")]
    // This attribute is needed to let the shell know that this package exposes some menus.
    [ProvideMenuResource("Menus.ctmenu", 1)]
    [ProvideService(typeof(ExpectLanguageService), ServiceName = "Expect Language Service")]
    [ProvideLanguageService(
            typeof(ExpectLanguageService),
            "Expect Language",
            106,            
            CodeSense = true,
            EnableCommenting = true,
            ShowCompletion = true,
            AutoOutlining = true,
            EnableAsyncCompletion = true,
            EnableLineNumbers = true
        )]
    [ProvideLanguageExtension(typeof(ExpectLanguageService), ".expect")]
    [ComVisible(true)]
    [ProvideAutoLoad(UIContextGuids80.SolutionExists)]
    [ProvideLoadKey("Standard", "1.0", "Expect Framework", "Expect", 1)]
    [ProvideService(typeof(SExpectVSService), ServiceName = "ExpectVSService")]
    [Guid(Guids.guidExpect_ExtensionsPkgString)]
    public sealed class ExpectPackage : Package, IOleComponent, IExpectPackage
    {
        private uint _componentID;

        private CommandSet _commandSet;

        private SelectionEvents _mySelectionEvents;

        private uint selectionEventsCookie;
        /// <summary>
        /// Default constructor of the package.
        /// Inside this method you can place any initialization code that does not require 
        /// any Visual Studio service because at this point the package object is created but 
        /// not sited yet inside Visual Studio environment. The place to do all the other 
        /// initialization is the Initialize method.
        /// </summary>
        public ExpectPackage()
        {
            Trace.WriteLine(string.Format(CultureInfo.CurrentCulture, "Entering constructor for: {0}", this.ToString()));
        }

        public object GetVSService(Type t)
        {
            return this.GetService(t);
        }        

        /////////////////////////////////////////////////////////////////////////////
        // Overriden Package Implementation
        #region Package Members

        /// <summary>
        /// Initialization of the package; this method is called right after the package is sited, so this is the place
        /// where you can put all the initilaization code that rely on services provided by VisualStudio.
        /// </summary>
        protected override void Initialize()
        {
            Trace.WriteLine (string.Format(CultureInfo.CurrentCulture, "Entering Initialize() of: {0}", this.ToString()));
            base.Initialize();
            try
            {
                (this as IServiceContainer).AddService(typeof(SExpectVSService), new ServiceCreatorCallback(CreateExpectVSService), true);

                IVsMonitorSelection selectionService = (IVsMonitorSelection)this.GetService(typeof(SVsShellMonitorSelection));
                _mySelectionEvents = new SelectionEvents(selectionService, this);
                selectionService.AdviseSelectionEvents(_mySelectionEvents, out selectionEventsCookie);

                _commandSet = new CommandSet(this);
                _commandSet.InitializeCommands((OleMenuCommandService)GetService(typeof(IMenuCommandService)));

                InitializeLanguageService();
            }
            catch (Exception ex)
            {
                WriteToActivityLog(__ACTIVITYLOG_ENTRYTYPE.ALE_ERROR, "Expect package throw an error\r\n" + ex.ToString());
                throw;
            }
        }

        private void WriteToActivityLog(__ACTIVITYLOG_ENTRYTYPE type, string msg)
        {
            IVsActivityLog log = GetService(typeof(SVsActivityLog)) as IVsActivityLog;
            if (log == null) return;

            int hr = log.LogEntry((UInt32)type, this.ToString(), msg);
        }

        private object CreateExpectVSService(IServiceContainer container, Type serviceType)
        {
            if (container != this)
            {
                return null;
            }

            if (typeof(SExpectVSService) == serviceType)
            {
                return new ExpectVsService(this);
            }

            return null;
        }

        private void InitializeLanguageService()
        {
            // Proffer the service.
            IServiceContainer serviceContainer = this as IServiceContainer;
            ExpectLanguageService langService = new ExpectLanguageService();
            langService.SetSite(this);
            serviceContainer.AddService(typeof(ExpectLanguageService),
                                        langService,
                                        true);

            // Register a timer to call our language service during
            // idle periods.
            IOleComponentManager mgr = GetService(typeof(SOleComponentManager))
                                       as IOleComponentManager;
            if (_componentID == 0 && mgr != null)
            {
                OLECRINFO[] crinfo = new OLECRINFO[1];
                crinfo[0].cbSize = (uint)Marshal.SizeOf(typeof(OLECRINFO));
                crinfo[0].grfcrf = (uint)_OLECRF.olecrfNeedIdleTime |
                                              (uint)_OLECRF.olecrfNeedPeriodicIdleTime;
                crinfo[0].grfcadvf = (uint)_OLECADVF.olecadvfModal |
                                              (uint)_OLECADVF.olecadvfRedrawOff |
                                              (uint)_OLECADVF.olecadvfWarningsOff;
                crinfo[0].uIdleTimeInterval = 1000;
                int hr = mgr.FRegisterComponent(this, crinfo, out _componentID);
            }

        }

        #endregion

        protected override void Dispose(bool disposing)
        {
            try
            {
                IVsMonitorSelection selectionService = (IVsMonitorSelection)this.GetService(typeof(SVsShellMonitorSelection));

                if (selectionService != null)
                {
                    selectionService.UnadviseSelectionEvents(selectionEventsCookie);
                }

                if (_componentID != 0)
                {
                    IOleComponentManager mgr = GetService(typeof(SOleComponentManager)) as IOleComponentManager;
                    if (mgr != null)
                    {
                        mgr.FRevokeComponent(_componentID);
                    }
                    _componentID = 0;
                }
            }
            finally
            {
                base.Dispose(disposing);
            }
        }

        #region IOleComponent Members
        public int FContinueMessageLoop(uint uReason, IntPtr pvLoopData, MSG[] pMsgPeeked)
        {
            return 1;
        }

        public int FDoIdle(uint grfidlef)
        {
            ExpectLanguageService ls = GetService(typeof(ExpectLanguageService)) as ExpectLanguageService;

            if (ls != null)
            {
                ls.OnIdle((grfidlef & (uint)_OLEIDLEF.oleidlefPeriodic) != 0);
            }

            return 0;
        }

        public int FPreTranslateMessage(MSG[] pMsg)
        {
            return 0;
        }

        public int FQueryTerminate(int fPromptUser)
        {
            return 1;
        }

        public int FReserved1(uint dwReserved, uint message, IntPtr wParam, IntPtr lParam)
        {
            return 1;
        }

        public IntPtr HwndGetWindow(uint dwWhich, uint dwReserved)
        {
            return IntPtr.Zero;
        }

        public void OnActivationChange(IOleComponent pic, int fSameComponent, OLECRINFO[] pcrinfo, int fHostIsActivating, OLECHOSTINFO[] pchostinfo, uint dwReserved)
        {
        }

        public void OnAppActivate(int fActive, uint dwOtherThreadID)
        {
        }

        public void OnEnterState(uint uStateID, int fEnter)
        {
        }

        public void OnLoseActivation()
        {
        }

        public void Terminate()
        {
        }
        #endregion
    }
}
